import tensorflow as tf

X = tf.Variable([[7, 0, -9], [0.55, -113, 12]], dtype=tf.float32);

atan = tf.atan(X)

d_atan = tf.gradients(atan, X)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    
    print('------atan------')
    print(sess.run(atan))
    
    print('------d_atan------')
    print(sess.run(d_atan))